package com.yjzx.llama.interfaces;

import com.yjzx.llama.tensor.FloatTensor;

/**
 * @author yjzx
 * @description Sampler
 * @since 2025/5/28
 */
@FunctionalInterface
public interface Sampler {
    int sampleToken(FloatTensor logits);

    Sampler ARGMAX = FloatTensor::argmax;
}
